from zipfile import ZipFile
import os


def getFiles(dir):
    file_paths = []
    for root, dirs, files in os.walk(dir):
        for filename in files:
            file_path = os.path.join(root, filename)
            file_paths.append(file_path)
    return file_paths

if __name__ == "__main__":
    dir = './screenshots'
    files = getFiles(dir)
    with ZipFile('reports.zip', 'w') as f:
        for file in files:
            f.write(file)
        f.close()
